package com.wang.gallerybackend.service.impl;

import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.util.ObjectUtil;
import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.wang.gallerybackend.constant.UserConstant;
import com.wang.gallerybackend.exception.BusinessException;
import com.wang.gallerybackend.exception.ErrorCode;
import com.wang.gallerybackend.exception.ThrowUtils;
import com.wang.gallerybackend.manager.auth.StpKit;
import com.wang.gallerybackend.model.dto.user.UserQueryRequest;
import com.wang.gallerybackend.model.dto.user.UserRegisterRequest;
import com.wang.gallerybackend.model.entity.User;
import com.wang.gallerybackend.model.enums.UserRoleEnum;
import com.wang.gallerybackend.model.vo.LoginUserVO;
import com.wang.gallerybackend.model.vo.UserVO;
import com.wang.gallerybackend.service.UserService;
import com.wang.gallerybackend.mapper.UserMapper;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import org.springframework.util.DigestUtils;

import javax.servlet.http.HttpServletRequest;
import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;

/**
 * @author 汪
 * @description 针对表【user(用户表)】的数据库操作Service实现
 *
 * */
@Service
@Slf4j
public class UserServiceImpl extends ServiceImpl<UserMapper, User>
		implements UserService {
	private static final String SALT = "wang";

	@Override
	public long userRegister(UserRegisterRequest userRegisterRequest) {
		//1.校验参数
		String userAccount = userRegisterRequest.getUserAccount();
		String userPassword = userRegisterRequest.getUserPassword();
		String checkPassword = userRegisterRequest.getCheckPassword();
		if (StrUtil.hasBlank(userAccount, userPassword, checkPassword)) {
			throw new BusinessException(ErrorCode.PARAMS_ERROR, "参数不能为空");
		}
		if (userAccount.length() < 4) {
			throw new BusinessException(ErrorCode.PARAMS_ERROR, "用户账号过短");
		}
		if (userPassword.length() < 8) {
			throw new BusinessException(ErrorCode.PARAMS_ERROR, "用户密码过短");
		}
		if (!userPassword.equals(checkPassword)) {
			throw new BusinessException(ErrorCode.PARAMS_ERROR, "两次输入的密码不一致");
		}
		//2.检查用户账号是否和数据库中已有的重复
		QueryWrapper<User> queryWrapper = new QueryWrapper<>();
		queryWrapper.eq("userAccount", userAccount);
		long count = this.baseMapper.selectCount(queryWrapper);
		ThrowUtils.throwIf(count > 0, ErrorCode.PARAMS_ERROR, "用户账号已存在");
		//3.密码加密
		String encryptPassword = getEncryptPassword(userPassword);
		//4.插入数据
		User user = new User();
		user.setUserAccount(userAccount);
		user.setUserPassword(encryptPassword);
		user.setUserName("无名");
		user.setUserRole(UserRoleEnum.USER.getValue());
		boolean save = this.save(user);
		if (!save) {
			throw new BusinessException(ErrorCode.SYSTEM_ERROR, "注册失败,数据库错误");
		}
		return user.getId();
	}

	/**
	 * 获取加密后的密码
	 * @param userPassword 密码
	 * @return 加密后的密码
	 */
	@Override
	public String getEncryptPassword(String userPassword) {
		// 加盐
		return DigestUtils.md5DigestAsHex((userPassword + SALT).getBytes());
	}

	@Override
	public LoginUserVO userLogin(String userAccount, String userPassword, HttpServletRequest request) {
		//1.校验
		ThrowUtils.throwIf(StrUtil.hasBlank(userAccount, userPassword), ErrorCode.PARAMS_ERROR, "参数不能为空");
		ThrowUtils.throwIf(userAccount.length() < 4, ErrorCode.PARAMS_ERROR, "用户账号错误");
		ThrowUtils.throwIf(userPassword.length() < 8, ErrorCode.PARAMS_ERROR, "用户密码错误");
		//2.对密码进行加密
		String encryptPassword = getEncryptPassword(userPassword);
		//3.查询用户是否存在
		QueryWrapper<User> queryWrapper = new QueryWrapper<>();
		queryWrapper.eq("userAccount", userAccount);
		queryWrapper.eq("userPassword", encryptPassword);
		User user = this.baseMapper.selectOne(queryWrapper);
		ThrowUtils.throwIf(user == null, ErrorCode.PARAMS_ERROR, "用户不存在或密码错误");
		//4.保存用户的登录态
		request.getSession().setAttribute(UserConstant.USER_LOGIN_STATE, user);
		// 记录用户登录态到sa-token
		StpKit.SPACE.login(user.getId());
		StpKit.SPACE.getSession().set(UserConstant.USER_LOGIN_STATE, user);
		return this.getLoginUserVO(user);
	}

	@Override
	public User getLoginUser(HttpServletRequest request) {
		//1.从session中获取当前登录用户
		User currentUser = (User) request.getSession().getAttribute(UserConstant.USER_LOGIN_STATE);
		//2.判断是否登录
		if (currentUser == null || currentUser.getId() == null) {
			throw new BusinessException(ErrorCode.NOT_LOGIN_ERROR);
		}
		//3.从数据库中查询用户信息
		Long userId = currentUser.getId();
		currentUser = this.getById(userId);
		if (currentUser == null) {
			throw new BusinessException(ErrorCode.NOT_LOGIN_ERROR);
		}
		return currentUser;
	}

	@Override
	public UserVO getUserVO(User user) {
		if (user == null) {
			return null;
		}
		UserVO userVO = new UserVO();
		BeanUtil.copyProperties(user, userVO);
		return userVO;
	}

	@Override
	public List<UserVO> getUserVO(List<User> userList) {
		if (CollUtil.isEmpty(userList)) {
			return new ArrayList<>();
		}
		return userList.stream()
				.map(this::getUserVO)
				.collect(Collectors.toList());
	}

	@Override
	public LoginUserVO getLoginUserVO(User user) {
		if (user == null) {
			return null;
		}
		LoginUserVO loginUserVO = new LoginUserVO();
		BeanUtil.copyProperties(user, loginUserVO);
		return loginUserVO;
	}

	@Override
	public boolean userLogout(HttpServletRequest request) {
		//1.从session中获取当前登录用户
		User currentUser = (User) request.getSession().getAttribute(UserConstant.USER_LOGIN_STATE);
		//2.判断是否登录
		if (currentUser == null) {
			throw new BusinessException(ErrorCode.OPERATION_ERROR, "未登录");
		}
		request.getSession().removeAttribute(UserConstant.USER_LOGIN_STATE);
		return true;
	}

	@Override
	public QueryWrapper<User> getQueryWrapper(UserQueryRequest userQueryRequest) {
		if (userQueryRequest == null) {
			throw new BusinessException(ErrorCode.PARAMS_ERROR, "请求参数为空");
		}
		Long id = userQueryRequest.getId();
		String userAccount = userQueryRequest.getUserAccount();
		String userName = userQueryRequest.getUserName();
		String userProfile = userQueryRequest.getUserProfile();
		String userRole = userQueryRequest.getUserRole();
		String sortField = userQueryRequest.getSortField();
		String sortOrder = userQueryRequest.getSortOrder();
		QueryWrapper<User> userQueryWrapper = new QueryWrapper<>();
		userQueryWrapper.eq(ObjectUtil.isNotNull(id), "id", id)
				.eq(StrUtil.isNotBlank(userRole), "userRole", userRole)
				.like(StrUtil.isNotBlank(userAccount), "userAccount", userAccount)
				.like(StrUtil.isNotBlank(userName), "userName", userName)
				.like(StrUtil.isNotBlank(userProfile), "userProfile", userProfile)
				.orderBy(StrUtil.isNotEmpty(sortField), "ascend".equals(sortOrder), sortOrder);
		return userQueryWrapper;
	}

	@Override
	public boolean isAdmin(User user) {
		return user != null && UserRoleEnum.ADMIN.getValue().equals(user.getUserRole());
	}
}




